Update search_graph.py