diff --git a/main.py b/main.py
index 4aacbed..f4c9b68 100644
--- a/main.py
+++ b/main.py
@@ -78,8 +78,9 @@ while True:
                             if interest in text:
                                 photos_saved += 1
                                 save_photo(image_url)
-
-            #sleep(1)
+            
+            query['completed'] = True
+            sleep(1)
     
     except KeyboardInterrupt:
         c.print(' Goodbye!', style='blue')
diff --git a/utils/authenticate.py b/utils/authenticate.py
index d7b3325..f2b9c17 100644
--- a/utils/authenticate.py
+++ b/utils/authenticate.py
@@ -1,6 +1,7 @@
 import vk_api
 import configparser
 from rich.console import Console
+from rich.prompt import Prompt
 c = Console()
 
 config = configparser.ConfigParser()
@@ -9,7 +10,7 @@ login = config['CREDENTIALS']['login']
 password = config['CREDENTIALS']['password']
 
 def _2fa_handler():
-    code = c.input("Enter 2FA code: ", style="italic cyan")
+    code = Prompt.ask("Enter 2FA code")
     return code, True
 
 def authenticate():