minor changes to plot