Slight change to plot function

slight change to plot()