Added back button and some styling fixes