Cleaned hyperparameter search code