Improve pruned tree model