Report best models while grid searching
Minor improvements on the grid search section