updated tests and docs