small change to plot script