Update UI for smaller screens