Update UnionFind.java