Adding plot and code for plot 2, minor fix for plot 1 code.