diff --git a/source/_static/css/custom.css b/source/_static/css/custom.css
index 8320c2cc4..a645e34bc 100644
--- a/source/_static/css/custom.css
+++ b/source/_static/css/custom.css
@@ -42,6 +42,28 @@ div.highlight:hover > pre::before {
   opacity: 0;
 }
 
+html[data-theme="light"] .highlight .gi { 
+  background-color: #ccffcc;
+  border: 1px solid #ccffcc;
+}
+
+html[data-theme="light"] .highlight .gd { 
+  background-color: #ffcccc;
+  border: 1px solid #ffcccc;
+  color: inherit !important;
+}
+
+html[data-theme="dark"] .highlight .gi { 
+  background-color: #005400;
+  border: 1px solid #005400;
+}
+
+html[data-theme="dark"] .highlight .gd { 
+  background-color: #540000;
+  border: 1px solid #540000;
+  color: inherit !important;
+}
+
 .topic {
   padding: 30px;
 }