Updated PyMC model building notebook