Fix a bug of data in testing set
Fix a bug of dataset forming