diff --git a/README.md b/README.md
index f501a07..d5cc31f 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # Grok-1
 
-This repository contains JAX example code for loading and running the Grok-1 open-weights model.
+This repository contains JAX example code for loading and running the Grok-1 open-weights model (XAI).
 
 Make sure to download the checkpoint and place the `ckpt-0` directory in `checkpoints` - see [Downloading the weights](#downloading-the-weights)