clean up sample project

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