add pruning for test data