Add support for screenshots with KIF