Return key when adding a value
Fix tests to expect add to return a key