diff --git a/README.md b/README.md
index 76a3701..12f40c6 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@ A frontend for large language models like [🐨 Koala](https://bair.berkeley.edu
 
 - Python 3.10
 - The pip packages listed in `requirements.txt`
-- A Koala model in the ggml format (should be quantized)
+- An AI model in the ggml format (should be quantized)
 
 For memory and disk requirements for the different models, see [llama.cpp - Memory/Disk Requirements](https://github.com/ggerganov/llama.cpp#memorydisk-requirements)
 
diff --git a/frontend/static/style.css b/frontend/static/style.css
index 49ad3a4..6a58b20 100644
--- a/frontend/static/style.css
+++ b/frontend/static/style.css
@@ -11,6 +11,7 @@
     --icon-button-fill: #ffffff;
     --send-icon-button-fill: #29c76d;
     --color: #fafafa;
+    --color2: #bbbbbb;
     --border-radius: .5rem;
 }
 
@@ -69,20 +70,38 @@ input[type="number"] {
 }
 
 .messages {
-    gap: 1.1rem;
+    gap: 1rem;
     margin-bottom: 1rem;
     overflow-y: scroll;
     max-height: 89vh;
+    align-items: center;
+    flex-grow: 2;
 }
 
 .message {
     display: flex;
-    flex-direction: row;
-    gap: .5rem;
-    padding: .5rem;
-    border-radius: var(--border-radius);
-    max-width: fit-content;
+    flex-direction: column;
+    flex-wrap: wrap;
+    gap: 1rem;
+}
+
+.message-type {
+    color: var(--color2);
+    text-align: center;
+}
+
+.message-text {
     white-space: pre-wrap;
+    padding: .5rem .8rem;
+    border-radius: var(--border-radius);
+}
+
+.message-bg-assistant > .message-text {
+    background: var(--background2);
+}
+
+.message-bg-user > .message-text {
+    background: var(--background3);
 }
 
 button {
@@ -115,19 +134,6 @@ button:hover {
     width: 100%;
 }
 
-.message-bg-assistant {
-    background: var(--background2);
-}
-
-.message-bg-user {
-    background: var(--background3);
-}
-
-.message-type {
-    min-width: 3.5rem;
-    padding-left: .1rem;
-}
-
 .input-container {
     margin-top: auto;
     flex-direction: row;
diff --git a/misc/screenshot.png b/misc/screenshot.png
index 9651407..0d97f75 100644
Binary files a/misc/screenshot.png and b/misc/screenshot.png differ
diff --git a/profiles/vicuna-v0.json b/profiles/vicuna-v0.json
index c62d575..17e0844 100644
--- a/profiles/vicuna-v0.json
+++ b/profiles/vicuna-v0.json
@@ -1,5 +1,5 @@
 {
-    "name": "Vicuna",
+    "name": "Vicuna v0",
     "conversation_prefix": "A chat between a curious human and a helpful AI assistant.\n\n",
     "user_keyword": "### Human:",
     "assistant_keyword": "### Assistant:",
diff --git a/profiles/vicuna-v1.1.json b/profiles/vicuna-v1.1.json
index 73829e3..c5b5b08 100644
--- a/profiles/vicuna-v1.1.json
+++ b/profiles/vicuna-v1.1.json
@@ -1,5 +1,5 @@
 {
-    "name": "Vicuna",
+    "name": "Vicuna v1.1",
     "conversation_prefix": "A chat between a curious user and a helpful AI assistant.\n\n",
     "user_keyword": "USER:",
     "assistant_keyword": "ASSISTANT:",