Update disjointset.py