Add StateMachine and Class graphs