Add unit tests to the library