impute_dataset function

Imputation to make a dataset complete