updated docs and test