First attempt – logistic regression