Added graph.py file