diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index d29927ef87297af0d710a371129a2543dd0e79a3..c806577b1a26fe1c2350de8c2d3efadb80e369fe 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -79,10 +79,13 @@ unit-test-job:   # This job runs in the test stage.
 pages:
   stage: deploy
   script:
-  - pip install -U sphinx m2r2 furo
+  - pip install -U sphinx m2r2 furo graphviz pyUML
   - pip install -e .
+  - cd docs/diagrams
+  - python class_diagram.py
+  - cd ../../
   - sphinx-build -b html docs/source public
-  - mv docs/diagrams ../
+  #- mv docs/diagrams ./public
   artifacts:
     paths:
     - public
diff --git a/docs/diagrams/balance.png b/docs/diagrams/balance.png
new file mode 100644
index 0000000000000000000000000000000000000000..0854c0d474f5b7b9f602f207f49c63216fad8a59
Binary files /dev/null and b/docs/diagrams/balance.png differ
diff --git a/docs/diagrams/folder_structure.png b/docs/diagrams/folder_structure.png
new file mode 100644
index 0000000000000000000000000000000000000000..4f09ef29b845e451138e86efeed082b55f5043c6
Binary files /dev/null and b/docs/diagrams/folder_structure.png differ