Adding regularization