Update social_graph.py