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.