Adding more integration test