adding gridsearch for initial model