UI changes to the screens