Update creategraph.py

cleaned up create graph