- Add conditional `display_screenshot` method