adding test and training sets.