Updating docs and fixing tests.