Update readDataSet.py