Add tests for the library
Add tests for the library