added support for returning to main menu and continuing game