Add tests to build