Update documentation and tests