add dedicated imputation class