Add media query for smaller screens