Small changes to plot 4