Added the MenuState class