ENH: better colorbar and vmin/vmax behavior