contextual class pruning small change