update code to include finish draft of dijkstra algorithm