diff --git a/README.md b/README.md
index c0065a9..6ba0e09 100644
--- a/README.md
+++ b/README.md
@@ -142,6 +142,8 @@ Finally, the main contribution of the paper. The repository offers the diffusion
 import torch
 from dalle2_pytorch import DiffusionPriorNetwork, DiffusionPrior, CLIP
 
+# get trained CLIP from step one
+
 clip = CLIP(
     dim_text = 512,
     dim_image = 512,