Add StateMachine impletements