Updated example in pymc script