The equation 49 is of the non-equilibrium formulation which is easier since it needs only the first order derivatives. However, my results don't match with benchmark results. In fact it is close to the homogeneous case only. The equations and numerical form are given in the attached image. Can you point where its wrong?