Update createDataSet.py