Add media query for mobile and window sizes