changed colorbars
changed to fig.colorbar
changed to plt.subplots()