Add support for screenshots