cleanup train data script