Add tests to the package