Added a class for Graph usage