Updated unit test and documentation.