Added a menu screen and state manager