Add a TODO to optimize the regularization model.