Update to remove the extra plot