Remade to StateBasedGame, added a main menu