Add some integration tests