diff --git a/frontend/static/styles.css b/frontend/static/styles.css
index 919732eb..6cb707d6 100644
--- a/frontend/static/styles.css
+++ b/frontend/static/styles.css
@@ -253,8 +253,8 @@ body .ddoc {
       @apply link;
     }
 
-    .highlight .context_button {
-      @apply border-none hover:bg-jsr-cyan-50;
+    pre.highlight .context_button {
+      @apply right-0 border-r-4 border-white hover:bg-jsr-cyan-50;
     }
 
     .highlight .lineNumbers {
@@ -266,6 +266,20 @@ body .ddoc {
     }
   }
 
+  .usageContent pre.highlight > code:first-child {
+    @apply pr-9;
+    display: flex;
+
+    * {
+      margin-left: 0.5rem;
+    }
+
+    span:first-child,
+    span:nth-child(4) {
+      margin-left: 0;
+    }
+  }
+
   .namespaceItemContent {
     flex: 1;
     width: 0;