Categorical variable importance now working