It would be good to use the current SaveScreenshot feature, to optionally assert that the screenshot taken matches a baseline specified screenshot.