Update and rename test.py to creation_dataset.py