Partial support for argmin() and argmax().

Currently only supports a single axis reduction at a time.
Remove blank last line in docstrings.
Cleanup and formatting.
Cleanup, formatting, and docstrings.