Maybe just call `plt.gray()` once (which sets the default colormap) and then leave off all the `cmap=plt.cm.gray` in the rest of the example.