Update docstrings and unit tests.