Cleaned up mainMenu() and added better code for changing gameState.