Improve UI in smaller screens