added a "back" button, and changed the layout