Added missing header comments into StateMachine