Get logistic regression working to check classification