Add Integration Tests