DOC dataset doc add return and improve doc #3315

Fixes #3315 
What do you think @arjoly ?