working on the imputation