Latest updated to plotting script