Lots of UI fixes

UI fixes and back button enable