add plot 2 code