Return the key on Put calls

This closer mimics the datastore API.