tools: add take_screenshot