fixed tree import to avoid tree.Tree