Add window screenshot functionality.