Update unitsGraph.py