added logistic regression, training model only