clean up impute stuff