Updated L2 loss