Added @media for smaller screen sizes