DOC dataset doc add return and improve doc #3315

Fixes #3315 
What do you think @arjoly ?
+1 for keeping "features" and stay consistent with the project naming conventions.