Add unit tests for the package
Add unit tests for the package