Add code for plot 1