Add media query for 320