Draw correct visited nodes in dijkstra's
Simplify dijkstra's a little