updating tests and docs