Layout improvements for media queries