add tag-wise pruning