def prepare_dataset