Create graph_search.py