modified python script to produce training/testing data
modified python script to produce training/testing data