add pruning