update for add graph base class