Add option for screenshot on failure