Added a plt.show() and switched the interpolation from 'none' to 'nearest' so that the code also works on OS X without warning messages.