more code for the logistic regression class