Fixed disappearing style issue
Added back button