add regularization to ML