Added missing line to code for second plot.