Add support for taking screenshots.