diff --git a/public/src/uncloseai-embed-modal.js b/public/src/uncloseai-embed-modal.js
index 9d9d53b..8477c66 100644
--- a/public/src/uncloseai-embed-modal.js
+++ b/public/src/uncloseai-embed-modal.js
@@ -93,18 +93,13 @@ async function getSelectedModel() {
 
 // Helper function to add copy and run buttons to code blocks
 function addCodeBlockCopyButtons(element) {
-	console.log("🔧 addCodeBlockCopyButtons called on element:", element);
 	const codeBlocks = element.querySelectorAll("pre code");
-	console.log("🔧 Found code blocks:", codeBlocks.length);
 	codeBlocks.forEach((codeBlock) => {
 		const pre = codeBlock.parentElement;
-		console.log("🔧 Processing code block, parent:", pre.tagName);
 		if (pre.tagName.toLowerCase() === "pre") {
 			// Check if buttons already exist to avoid duplicates
 			const existingBtn = pre.querySelector('.uncloseai-code-copy-btn');
-			console.log("🔧 Existing button check:", existingBtn);
 			if (existingBtn) {
-				console.log("🔧 Buttons already exist, skipping");
 				return;
 			}
 
@@ -163,7 +158,11 @@ function addCodeBlockCopyButtons(element) {
 					resultsContainer.className = 'uncloseai-code-execution-results';
 					resultsContainer.style.marginTop = '10px';
 					resultsContainer.style.padding = '10px';
-					resultsContainer.style.backgroundColor = 'var(--uncloseai-modal-background)';
+
+					// Match the code block's background color
+					const preStyles = window.getComputedStyle(pre);
+					resultsContainer.style.backgroundColor = preStyles.backgroundColor;
+
 					resultsContainer.style.borderRadius = '5px';
 					resultsContainer.style.fontFamily = 'monospace';
 					resultsContainer.style.fontSize = '14px';