Restoring plot2.R with Project 1 version