Edited the css and fixed the back button