Impute2 small update