Updated the logic of feature pruning based on notes from the Refex paper.