Implementation with img_as_ubyte instead of convert, reformatting of comments, use of plt.subplots instead of plt.subplot and removed repeatitive package declarations