Add logistic regression and grid search