Should we use just promises?
I mean, instead of calling the callback just return the promise.

Not blocking here, but I would like we move to start using promises, and here looks perfect since datastore is already using them.