Updates unit tests and docstrings.