Small changes to plot