diff --git a/src/components/code-block.js b/src/components/code-block.js
index 939857b..31fca27 100644
--- a/src/components/code-block.js
+++ b/src/components/code-block.js
@@ -1,39 +1,17 @@
-// @ts-check
-
 class CodeBlockComponent extends HTMLElement {
-  constructor() {
-    super();
-  }
-
   connectedCallback() {
-    const trimmed = this.innerHTML.trim();
+    this.trimmed = this.innerHTML.trim();
+    const lang = this.getAttribute("language");
+    const lineNumbers = this.getAttribute("line-numbers") === "true";
 
-    this.css({
-      color: "orange",
-      margin: "1em 0",
-      padding: "1em",
-      fontWeight: "bold",
-      fontSize: "x-large",
-      display: "block",
-      backgroundColor: "#eee",
-    });
+    console.log(lineNumbers, lang);
 
-    const template = `
+    this.innerHTML = `
       <div>
-        <pre class="language-${this.lang}" id="pre"><code id="code">${trimmed}</code></pre>
+        <pre class="language-${lang} ${
+      lineNumbers ? "line-numbers" : ""
+    }" id="pre"><code id="code">${this.trimmed}</code></pre>
       </div>
     `;
-
-    this.innerHTML = template;
-  }
-
-  /**
-   * Apply css to an element
-   * @param {Partial<CSSStyleDeclaration>} style
-   * @param {HTMLElement} element
-   */
-  css(style, element = null) {
-    let el = element ?? this;
-    for (const property in style) el.style[property] = style[property];
   }
 }
diff --git a/src/pages/demo/index.html b/src/pages/demo/index.html
index dc66031..a1289aa 100644
--- a/src/pages/demo/index.html
+++ b/src/pages/demo/index.html
@@ -1,6 +1,8 @@
 <!DOCTYPE html>
 <html lang="en">
   <my-head>
+    <link rel="stylesheet" href="/prism.css" />
+    <script src="/prism.js"></script>
     <script server:setup>
       const name = "AYOs";
       let greeting = "hello";
@@ -25,7 +27,12 @@
       <p>some text: {{greeting}}</p>
       {{greeting}} hey<br />
       <clickable-text></clickable-text>
-      <code-block lang="css"> const hey = "AYO" </code-block>
+      <code-block language="js" line-numbers="true">
+        <pre>
+const hey = "AYO";
+console.log(hey)
+        </pre>
+      </code-block>
     </main>
     <my-footer></my-footer>
 
