Fixing gridsearch script