Update dataset_train_test.m