Added back button. Cleaning code.