small changes for tree