Added "Exit" button to menu, cleaned up code.