add integration testing