diff --git a/setup_diagrams.sh b/setup_diagrams.sh
index ce4daec..8b1bbe3 100644
--- a/setup_diagrams.sh
+++ b/setup_diagrams.sh
@@ -53,4 +53,7 @@ apt install --reinstall libexpat1-dev libexpat1
 pip3 install flask
 
 # cd /var/diagrams && curl -L http://sourceforge.net/projects/plantuml/files/plantuml.jar/download -o plantuml.jar
-# apt install mscgen
\ No newline at end of file
+# apt install mscgen
+
+# Production Flask
+pip install waitress