Added Integration Test