add support of categorical features