diff --git a/README.md b/README.md
index 6c000cd0..cd6c20a9 100644
--- a/README.md
+++ b/README.md
@@ -68,6 +68,7 @@ from torch import nn
 class My_Model(RootModule):
     def __init__(self):
         # define model
+        self.l1 = nn.Linear(200, 10)
     
     # ---------------
     # TRAINING