diff --git a/README.md b/README.md
index 6b7ae55..3283b15 100644
--- a/README.md
+++ b/README.md
@@ -36,3 +36,11 @@ Initiate the pipeline
 ```python
 python main.py
 ```
+
+### Unit Tests
+
+To run the unit tests, run the command below.
+
+```bash
+pytest tests/test_file_name.py
+```
diff --git a/architecture.png b/architecture.png
index 9e622c8..d2f91a8 100644
Binary files a/architecture.png and b/architecture.png differ