Working on building logistic regression model