Rewriting the menu class, and the game in general