Update GraphSearch.py