Small changes to script for sampling with PyMC.