Add L2 regularization in the code!