- added l2 regularization