Add media query (<520px)