Add full tangent stiffness function