diff --git a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor
index 75877da0..51aca3a5 100644
--- a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor	
+++ b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor	
@@ -1,7 +1,7 @@
 @using AIStudio.Tools
 @using MudBlazor
 
-<MudCard Class="my-2 rounded-lg" Outlined="@true">
+<MudCard Class="@this.CardClasses" Outlined="@true">
     <MudCardHeader>
         <CardHeaderAvatar>
             <MudAvatar Color="@this.Role.ToColor()">
diff --git a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs
index a08c5bfa..f411b075 100644
--- a/app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs	
+++ b/app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs	
@@ -33,6 +33,12 @@ public partial class ContentBlockComponent : ComponentBase
     [Parameter]
     public DateTimeOffset Time { get; init; }
     
+    /// <summary>
+    /// Optional CSS classes.
+    /// </summary>
+    [Parameter]
+    public string Class { get; set; } = string.Empty;
+    
     [Inject]
     private Rust Rust { get; init; } = null!;
     
@@ -107,4 +113,6 @@ public partial class ContentBlockComponent : ComponentBase
                 break;
         }
     }
+    
+    private string CardClasses => $"my-2 rounded-lg {this.Class}";
 }
\ No newline at end of file