Adding the code to get categories based on ids, as well as the logistic regression work