Logistic regression seems to work