Add screen capture functions