Update for PyStan