Update to Flag v2 API