Finish docstring and add test