add regularization