added media query for small screens