Adding future integration tests