Update to plot3

Fixes legend
Update to plot 4

Fixes legend issue