Update dijkstra.py

Using a new algorithm