modified:   infectGraph.py