Update dijkstra.py