Update StateMachine.h