Added Integration test