Updating Contains to properly track what is going on in the WriteBatch