Add graph attribute on StateMachine