change toy problem hyperparameters
prettier plotting