divided test and train data