update tests and docs