modified imputation scripts