Added more UnionFind tests, they all pass :)