updated_dataset = dataset.map(lambda example: {"question": example["query"]["en"]}, remove_columns=["query"])
updated_dataset = updated_dataset.map(
     lambda example: {"answer": example["answers"][0]}, remove_columns=["answer", "answers"]
 )