Logistic regression minor changes