Remove main
Implement dijkstra