Update GenerateSearchResultDataset.java