Added Integration tests