<human>:  How to add a colorbar for a hist2d plot
<bot>: Here is some code to add a colorbar to plot `plt` using image `im` on axes `ax`

plt.colorbar(im, ax=ax)

#Warning: this code generation is experimental. Please inspect the code for any bugs before executing.