Add media query to detect screen size