Adding Logistic Regression Code