expand imputation lab
change code slightly