Add initial integration tests