Update shortest_path.py