Add code for plot1 and plt2