UI changes on main screen