Update grid_search.py