update graph_search for tidying