def process_docs(dataset):
    def _helper(doc):
        doc["choices"] = [doc["Option A"], doc["Option B"], doc["Option C"]]
        return doc

    return dataset.map(_helper)