Small adjustment to plot4