Shortest Path Algorithm Python