dijkstra's algorithm code snippets in python