updated test and docs