Dijkstra with priority queue python