Create graph.py revised