update to plotting script