changed imputation method, added more narrative