modify plot code

modify plot code