Improve pruning