add logistic regression?