Added a tree generator
improved the tree