Modified game state machine slightly