Add media query for different sizes