Added forward declaration in StateMachine