replace convergence with rhat
can now pass data to plot_residuals for subsets of data