updates to test and docs