Made small edits to plot4