Improve imputation explanation