Update graphGenerator.py