add plot code and output