A little additional tweaking to feature pruning