add WithScreenshotOnError to TODO