Modified plotter.py file