more improvements to UI for smaller screens