Modified the JSON schema to have a control flow