Changed the python script for generating the pngs