Add tests to test the integration of data