Modify vertical diffusion of total energy

Currently, we are diffusing total enthalpy in the total energy equation. This is not accurate, and we should modify it following the dycore paper

<img width="442" height="49" alt="Image" src="https://github.com/user-attachments/assets/216d75a2-acd1-4eba-9757-4e8a371026a5" />