For the classes, I'd like to switch to endog, exog to follow the pattern for a regression model