Fixing up tests and documentation.