Augment state object with get fn