Add media query for mobile