minor changes to dijkstra.py