Update screenshot.py

screenshot.py

@@ -1,3 +1,4 @@
+from selenium import webdriver
 from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By