Prevent the lngrad cross from blowing up in case the background
stratification is exactly 0
Added rayleigh taylor instability as a test case