diff --git a/src/components/thread/index.tsx b/src/components/thread/index.tsx
index a1fa7e9..2332f30 100644
--- a/src/components/thread/index.tsx
+++ b/src/components/thread/index.tsx
@@ -315,7 +315,7 @@ export function Thread() {
 
                 <ScrollToBottom className="absolute bottom-full left-1/2 -translate-x-1/2 mb-4 animate-in fade-in-0 zoom-in-95" />
 
-                <div className="bg-muted rounded-2xl border mx-auto mb-8 w-full max-w-4xl relative z-10">
+                <div className="bg-muted rounded-2xl border shadow-xs mx-auto mb-8 w-full max-w-4xl relative z-10">
                   <form
                     onSubmit={handleSubmit}
                     className="grid grid-rows-[1fr_auto] gap-2 max-w-4xl mx-auto"
@@ -344,7 +344,7 @@ export function Thread() {
                       ) : (
                         <Button
                           type="submit"
-                          className="transition-all"
+                          className="transition-all shadow-md"
                           disabled={isLoading || !input.trim()}
                         >
                           Send
diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index f606c1a..857085c 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -69,15 +69,12 @@ export function AssistantMessage({
 
   return (
     <div className="flex items-start mr-auto gap-2 group">
-      <Avatar>
-        <AvatarFallback>A</AvatarFallback>
-      </Avatar>
       {isToolResult ? (
         <ToolResult message={message} />
       ) : (
         <div className="flex flex-col gap-2">
           {contentString.length > 0 && (
-            <div className="rounded-2xl bg-muted px-4 py-2">
+            <div className="py-1">
               <MarkdownText>{contentString}</MarkdownText>
             </div>
           )}
diff --git a/src/components/thread/messages/human.tsx b/src/components/thread/messages/human.tsx
index 340a634..7c0b553 100644
--- a/src/components/thread/messages/human.tsx
+++ b/src/components/thread/messages/human.tsx
@@ -84,7 +84,7 @@ export function HumanMessage({
             onSubmit={handleSubmitEdit}
           />
         ) : (
-          <p className="text-right py-1">{contentString}</p>
+          <p className="text-right px-4 py-2 rounded-3xl bg-muted">{contentString}</p>
         )}
 
         <div