Updates to plot4