diff --git a/README.md b/README.md
index fdcbfb4..ebc9dc4 100644
--- a/README.md
+++ b/README.md
@@ -24,7 +24,7 @@ Below is a recap of the main changes made since:
 
 ## Installation
 
-```
+```sh
 $ go get github.com/jessp01/gohighlight
 ```
 
diff --git a/syntax_files/go.yaml b/syntax_files/go.yaml
index 95c5a51..480a162 100644
--- a/syntax_files/go.yaml
+++ b/syntax_files/go.yaml
@@ -2,6 +2,7 @@ filetype: go
 
 detect:
     filename: "\\.go$"
+    header: "^package\\s.*"
 
 rules:
     # Conditionals and control flow