add plot 4 code