fixed hyperparameters and have a plotting code