Add tests to project