update tests and documentation