Fix package names.
Add missing docstring.
Add tests.