Scala/Cucumber - take a screenshot after each step

gistfile1.txt

class Hooks extends ScalaDsl with EN with BrowserDriver {
  AfterStep { scenario: Scenario =>
    if (scenario.isFailed) {
      val screenshotName = scenario.getName.replaceAll(" ", "_")
      val screenshot     = driver.asInstanceOf[TakesScreenshot].getScreenshotAs(OutputType.BYTES)
      scenario.attach(screenshot, "image/png", screenshotName)
    }
  }
}