fix plotting bug in run script
testing with old method first