Add the class to generate graphs