Added test and train data.