Updated with plot 1 & 2 codes