diff --git a/.gitignore b/.gitignore
index ba35e2a..f796c36 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,5 @@ config.ini
 screenshots
 downloaded_files
 __pycache__
+
+secrets.ini
diff --git a/screenshots.py b/screenshots.py
index c81ac0b..328b3ef 100644
--- a/screenshots.py
+++ b/screenshots.py
@@ -79,6 +79,13 @@ def take_screenshot(tracker_name, driver=driver, is_load_at_runtime=False):
 
     # Login and save screenshot
     driver.find_element(By.CLASS_NAME, "auth-form__primary-button").click()
+
+    # 2FA is enabled, ask for the code
+    if driver.find_elements(By.ID, "code"):
+        code = input("Please enter the 2FA code: ")
+        code_field = driver.find_element(By.ID, "code")
+        code_field.send_keys(code)
+
     take_screenshot("Aither")
     print("Aither Screenshoted")