Add imputation strategy