Add media query for smaller screens.