Add union-find data structure