Added MenuState functionality