Got rid of ScreenStateMachine