Logistic regression code added