update for grid search with features