Add media queries for smaller screens