diff --git a/README.md b/README.md
index 86b6748ee016a27e77d8b4477baacbe94343520c..3a449d6e32ceb063ab59fac404d53928ca7b8698 100644
--- a/README.md
+++ b/README.md
@@ -63,6 +63,20 @@ This project is setup as a package which means you can now easily import any fil
 ```python
 from project.datasets.mnist import mnist
 from project.lit_classifier_main import LitClassifier
+from pytorch_lightning import Trainer
+
+# model
+model = LitClassifier()
+
+# data
+train, val, test = mnist()
+
+# train
+trainer = Trainer()
+trainer.fit(model, train, val)
+
+# test using the best model!
+trainer.test(test_dataloaders=test)
 ```
 
 ### Citation