Check formulation first in Pedlosky 1984 and Grooms et al. 2012. One should use a spatially variable stratification in principle