Add pruning