training a tree, and will add the pruning code