java plugin: Change the API to use the `DataSet' class.