Improve style for small screens