DOC: better doc for impurity_improvement