logistic regression added , it was awesome