Integrate rate heterogeneity from python codes
Integrate rate heterogeneity from python codes