Make `set` chainable

`.set` should return self, according to the spec https://people.mozilla.org/~jorendorff/es6-draft.html#sec-weakmap.prototype.set