`Kd_add` is added to `Kd` after we diagnose the work (Kd * N2 * dz): https://github.com/NOAA-GFDL/MOM6/blob/31a4d8b4576e4696b6add76fe1ef31638849e284/src/parameterizations/vertical/MOM_set_diffusivity.F90#L671 shouldn't we invert the two so we diagnose the total work ?