Add unit tests and docstring