add pruning function