UI changes

UI changes to the app