Add integration testing