Clean up code for plot3.png