Improving a land surface scheme for estimating sensible and latent heat fluxes above grasslands with contrasting soil moisture zones