Update write.py

Using 'data-set' in comment