More tests.
Updated docstrings.