Add Integration Test