Reimplementation of the imputation algorithm using the lme4 package with modification to the returned object