Add tests to package