adding test and train data