Updated plot method