Add `set` method