Impute with categorical variables