Update union_find.cpp