Add plot1 code