Added the tree class in trial