diff --git a/mkdocs.yml b/mkdocs.yml
index 539714d4..157f7d92 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -8,3 +8,8 @@ site_description: 'Documentation for PyTorch LightningModule, the researcher ver
 
 dev_addr: '0.0.0.0:8000'
 #google_analytics: ['UA-aasd', 'sitename']
+
+markdown_extensions:
+  - codehilite:
+      guess_lang: false
+      linenums: true