Editing plot code