FIX: impurity_improvement