diff --git a/bot/handlers/commands.py b/bot/handlers/commands.py
index 102861d..c1be10f 100644
--- a/bot/handlers/commands.py
+++ b/bot/handlers/commands.py
@@ -1,5 +1,5 @@
 from aiogram import Router
-from aiogram.types import Message, FSInputFile
+from aiogram.types import Message
 from aiogram.filters import Command
 
 
diff --git a/bot/handlers/inline.py b/bot/handlers/inline.py
index afa1bcc..ca11580 100644
--- a/bot/handlers/inline.py
+++ b/bot/handlers/inline.py
@@ -1,10 +1,9 @@
-from typing import Dict, Any
+from typing import Any
 from time import monotonic
 
 from aiogram import Router, F
 
 from aiogram.types import CallbackQuery
-from aiogram.filters.callback_data import CallbackData
 from aiogram.fsm.storage.memory import MemoryStorage
 
 router = Router()
diff --git a/bot/middleware.py b/bot/middleware.py
index f7371da..392cede 100644
--- a/bot/middleware.py
+++ b/bot/middleware.py
@@ -20,14 +20,11 @@ async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[
         
         if not user_id in my_storage:
             my_storage[user_id] = [time, False]
-            # print(1)
         elif my_storage[user_id][1]:
-            # print(2)
             return
         elif my_storage[user_id][0] + .5 > time: # new message sent less than in 0.5 sec
             my_storage[user_id] = [time, True]
             await event.answer("You Sooooo Fast!!! Now You Should Make Captcha Below To Continue", reply_markup=captcha_inline())
-            # print(3)
             return
         else:
             my_storage[user_id][0] = time
diff --git a/main.py b/main.py
index 0f23fc6..d5d5add 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,7 @@
 import logging
 import os
 
-from aiogram import Dispatcher, types, Bot, F
+from aiogram import Dispatcher, Bot, F
 from aiogram.fsm.storage.memory import MemoryStorage
 
 from bot import middleware
@@ -32,10 +32,10 @@ async def run():
     # Middleware here
     dp.message.middleware(middleware.AntiFloodMiddleware())
 
-    # try:
-    await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types())
-    # finally:
-    #     await bot.session.close()
+    try:
+        await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types())
+    finally:
+        await bot.session.close()
 
 if __name__ == "__main__":
     asyncio.run(run())
\ No newline at end of file