Update create_dataset.py

comments