Improve docstring for dataset.Dataset