Add generate_drive_dataset