update kruskal_min_spanning_tree.py