Minor amendments to plot2 code