Clean up and added comments for main menu to game transition