Added the support for categorial columns