Added partition data script to make test and training set