changes for grid search