Clean up Dataset and DatasetDict
_The documentation is not available anymore as the PR was closed or merged._
This PR cleans up the docstrings for the other half of the methods in `Dataset` and finishes `DatasetDict`.