Add tests to build process