Added decision tree pruning.