Work on the the layout for small screens