Update plots to copy directly to device

Helps with the legend printing in plot 3 particularly.