I have now added the ability to create color density plots
using the plt.scatter and choosing the right pixel size.
A color bar has been added and the figure namings are now
automated by adding '.png' extension to the source file.