Added NA handling to `dispersion_plot`