Added some more integration test cases