Added media query for small screens