Add a Graph Generator class