Added L2 penalty with weight 0.001