method to highlight webelement in yellow color

HighlightWebElement.java

public void highlight(WebElement element, long highlightInterval) throws InterruptedException {
        int flexibleWait = 5;
        long pollingInterval = 500;
        WebDriverWait wait = new WebDriverWait(ngDriver, flexibleWait);
        wait.pollingEvery(pollingInterval, TimeUnit.MILLISECONDS);
        wait.until(ExpectedConditions.visibilityOf(element));
        executeScript("arguments[0].style.border='3px solid yellow'", element);
        Thread.sleep(highlightInterval);
        //executeScript("arguments[0].style.border=''", element);
    }