Code I am using to learn PyMC