Shortest Path Python