How do you take a screenshot for a failed scenario in Cucumber?

package hooks;

import io.cucumber.java.After;
import io.cucumber.java.Scenario;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;

public class Hooks {

    WebDriver driver = DriverManager.getDriver(); // get from ThreadLocal

    @After
    public void closeBrowser(Scenario scenario) {
        if (scenario.isFailed()) {
            // Capture screenshot as byte array
            byte[] screenshot = ((TakesScreenshot) driver)
                .getScreenshotAs(OutputType.BYTES);

            // Attach to Cucumber HTML report
            scenario.attach(screenshot, "image/png", "Failure Screenshot");
        }

        if (driver != null) {
            driver.quit();
        }
    }
}

@After
public void closeBrowser(Scenario scenario) {
    if (scenario.isFailed()) {
        // Save to file
        File src = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        String path = "screenshots/" + scenario.getName().replaceAll(" ", "_") + ".png";
        FileUtils.copyFile(src, new File(path));

        // Also embed in report
        byte[] bytes = ((TakesScreenshot) driver).getScreenshotAs(OutputType.BYTES);
        scenario.attach(bytes, "image/png", scenario.getName());
    }
    driver.quit();
}

@After
public void closeBrowser(Scenario scenario) {
    if (scenario.isFailed()) {
        // write code for screenshot
        byte[] screenshot = ((TakesScreenshot) driver)
            .getScreenshotAs(OutputType.BYTES);
        scenario.embed(screenshot, "image/png");  // deprecated in v5+
    }
}

public class Hooks {

    @Before
    public void setUp(Scenario scenario) {
        // Initialize driver before scenario
        System.out.println("Starting: " + scenario.getName());
        DriverManager.initDriver("chrome");
    }

    @After
    public void tearDown(Scenario scenario) {
        if (scenario.isFailed()) {
            byte[] screenshot = ((TakesScreenshot) DriverManager.getDriver())
                .getScreenshotAs(OutputType.BYTES);
            scenario.attach(screenshot, "image/png", "Failure Screenshot");
        }
        DriverManager.quitDriver();
    }
}