Update network_graph.py