Fix the plot_obs_pred function so that it property handles the rarity bins