Added a gridsearch & shuffle the data for SGD