Take screenshots on test failure

Take a screenshot on test failure or error.
@jzoldak