Add media query for smaller screen sizes