add non Conservation flux term

use non conservation flux term to balance the variability of depth