Added media query for smaller screens