updated to new API
updated tests
docstring update