Add tests to the project