Update docstring
Add new tests