Added docstring for ax_colorbar
More docstring updates.