Add regularization