Create disjoint_set.py