Add regularization framework + l2 regularization