Put l2 regularizer back in