Add retrain to GridSearch