Update zipTrainData.py