updated the tests and docs