Added error handling, comments and exit command to menu.