Update primKruskalDijkstra.py