Add media query for max-width: 768