modified plot_posterior.py