DOC dataset doc add return and improve doc #3315

Fixes #3315