Update kruskal.py