diff --git a/doc-generator/src/highlighter/mod.rs b/doc-generator/src/highlighter/mod.rs
index d37e9d4..b1d8d37 100644
--- a/doc-generator/src/highlighter/mod.rs
+++ b/doc-generator/src/highlighter/mod.rs
@@ -53,6 +53,17 @@ pub fn highlight(input: &String) -> String {
 
                 output.replace_range(range, html.as_str());
             }
+            TokenType::Comment => {
+                let start_pos = token.position;
+                let end_pos = token.get_end_position();
+
+                let range = (start_pos + offset)..(end_pos + offset);
+                let html = format!("<span class=\"token comment\">{}</span>", token.value);
+
+                offset += 35;
+
+                output.replace_range(range, html.as_str());
+            }
             _ => {}
         }
     }