Add code to plot the posterior probabilities