add grid search stuff