Updates to Union/Find module