edit python plotting script