created logistic regression model in the main method