fixed bug that was changing the tree array after the first time
tmp.union() was getting called
improved the way self.sz is initiated