Update documentation. Small cleanup in test cases.