add partial prior predictive distribution code