Added integration test