Adding logistic regression class