Added the Graph code