Add more tests to integration suite