diff --git a/README.md b/README.md
index b84adf0..dd2db84 100644
--- a/README.md
+++ b/README.md
@@ -84,4 +84,11 @@ prakhars962/deepinn   <none>        fe16ca34f9d9   About an hour ago   6.8GB
 The only solution is to delete them one by one using the IMAGE_ID.
 ```sh
 docker image rm -f IMAGE_ID
+```
+
+### UML diagram
+The UML diagram is generated using `pyreverse`. This provides an overview of the codebase.
+
+```sh
+pyreverse -o png DeepINN/
 ```
\ No newline at end of file
diff --git a/classes.png b/classes.png
new file mode 100644
index 0000000..0bc66be
Binary files /dev/null and b/classes.png differ
diff --git a/packages.png b/packages.png
new file mode 100644
index 0000000..2c93fc4
Binary files /dev/null and b/packages.png differ
diff --git a/todo.md b/todo.md
index 40dc473..1cf9d25 100644
--- a/todo.md
+++ b/todo.md
@@ -17,7 +17,7 @@ Last work : `DeepINN/constraint/gradients.py`
 - [X] Implement PDE loss constraint.
 - [x] Implement gradients.
 - [ ] Implement lazy evaluation of gradients.
-- [ ] Implement more constrainst.
+- [ ] Implement more constraint.
 
 ## Architectures
 - [x] Implement fully connected NN.