diff --git a/app/styles/output.css b/app/styles/output.css
index c5db2b3..09e1f09 100644
--- a/app/styles/output.css
+++ b/app/styles/output.css
@@ -53,9 +53,21 @@
     color: var(--text-color);
 }
 
-.ai-message table, th, tr, td{
-    border: 2px solid black;
+.ai-message table{
+    border: 1px solid var(--ai-message-text-color);
     border-collapse: collapse;
+    background-color: var(--conversation-background-color);
+}
+
+.ai-message th{
+    background-color: var(--user-message-background-color);
+    align-items: center;
+    padding: 5px;
+}
+
+.ai-message td{
+    align-items: center;
+    padding: 5px;
 }
 
 .ai-message img{
@@ -66,6 +78,15 @@
     filter: brightness(70%);
 }
 
+.ai-message li{
+    margin-left: 1em;
+}
+
+.ai-message code{
+    width: 100%;
+    overflow-wrap: break-word;
+}
+
 /* Button Container */
 .button-container {
     display: flex;