diff --git a/static/css/codeblock.css b/static/css/codeblock.css
index 36fcdb7..b4468c1 100644
--- a/static/css/codeblock.css
+++ b/static/css/codeblock.css
@@ -1,9 +1,33 @@
 /* --- Code blocks --- */
 
-.chroma .ln { 
-  margin-right: 0.8em; 
-  padding: 0 0.4em 0 0.4em; 
+.chroma .ln {
+  margin-right: 0.8em;
+  padding: 0 0.4em 0 0.4em;
 }
 pre code.hljs {
   padding: 9.5px;
 }
+
+.highlight tr, .highlight pre {
+  border: none;
+}
+
+.highlight div:first-child {
+  border-radius: 4px;
+}
+
+.highlight td:first-child pre {
+  border-top-left-radius: 4px;
+  border-top-right-radius: unset;
+  border-bottom-left-radius: 4px;
+  border-bottom-right-radius: unset;
+  overflow: hidden;
+}
+
+.highlight td:last-child pre {
+  border-radius: unset;
+}
+
+.highlight td:last-child pre code {
+  white-space: pre;
+}