Update to include notify