added class to draw graphs