test and train data loaded