some changes for hyperparameter optimization