Switched highlighting from the <pre> elements to the <code> elements

Author: alexiglesias93

_code-blocks/webflow/embed-code.js

@@ -31,7 +31,7 @@ document.addEventListener('DOMContentLoaded', async () => {
     preElement.appendChild(copyButton);
 
     // Highlight the element
-    highlightJS.highlightBlock(preElement);
+    highlightJS.highlightBlock(codeElement);
 
     // Append it to the embed
     embedElement.appendChild(preElement);

_code-blocks/webflow/embed-code.ts

@@ -30,7 +30,7 @@ document.addEventListener('DOMContentLoaded', async () => {
     preElement.appendChild(copyButton);
 
     // Highlight the element
-    highlightJS.highlightBlock(preElement);
+    highlightJS.highlightBlock(codeElement);
   }
 
   initCopyClipboard();