Update creation_dataset.py