Updating docs with optional `context` arg for a few methods.
`groupBy`, `countBy` and `sortedIndex` can all take an optional
`context` argument, but this wasn't in the docs.
Merge pull request #947 from lukeasrodgers/add_context_param_to_docs

Updating docs with optional `context` arg for a few methods.