edits to plotting program