Updating documentation,
Supporting categorical variables in the high level api.