Adding an integration test