add trainSet and testSet