updates to plotting script