diff --git a/README.md b/README.md
index 068ce57..3465d4b 100644
--- a/README.md
+++ b/README.md
@@ -10,7 +10,7 @@
 
 **Polymathic's Large Omnimodal Model for Astronomy**
 
-[🚀 Quick Start](#-quick-start) • [🔬 Scientific Overview](#-scientific-overview) • [📦 Advanced Installation](#-advanced-installation) 
+[🚀 Quick Start](#-quick-start) • [🔬 Scientific Overview](#-scientific-overview) • [📦 Advanced Installation](#-advanced-installation)
 
 </div>
 
@@ -71,10 +71,9 @@ embeddings = model.encode(tokens, num_encoder_tokens=600)
 
 # Option 2: Generate predictions (e.g., redshift)
 from aion.modalities import Z
-predictions = model(
-    tokens,
-    target_mask={'tok_z': torch.zeros(batch_size, 1)},
-    num_encoder_tokens=600
+preds = model(
+    codec_manager.encode(image),
+    target_modality=Z,
 )
 ```