A nice wrapper for Dijkstra's shortest path algorithm in Python