added PyMC3 code