UI changes for main and other screens