More docstrings and unit tests. Removed a redundant method arg.