COSMITS + FIX to impurity_improvement