add prune for Tree Regression