Updated plot2.R code