Added back button styling and small ux changes