Added the ui changes

UI changes made by George