diff --git a/extension/content.js b/extension/content.js
index 4f42905..91fd45f 100644
--- a/extension/content.js
+++ b/extension/content.js
@@ -33,6 +33,17 @@
                 document.querySelector('div[class*="question-content"]');
             const description = descriptionElement ? descriptionElement.innerText : "No description found.";
 
+            // Extract difficulty badge
+            const difficultyElement =
+                document.querySelector('.difficulty') ||
+                document.querySelector('.text-difficuly-easy') ||
+                document.querySelector('.text-difficuly-medium') ||
+                document.querySelector('.text-difficuly-hard');
+
+            const difficulty = difficultyElement
+                ? difficultyElement.innerText.trim()
+                : "Unknown Difficulty";
+
             let code = "";
             const viewLines = document.querySelector('.view-lines');
             if (viewLines) {