Square boxes layout with container queries