Updating documentation and fixing test.