Added plot code and new comments