diff --git a/report/report.tex b/report/report.tex
index 6c21a56..6ae674f 100644
--- a/report/report.tex
+++ b/report/report.tex
@@ -122,8 +122,7 @@
 			\end{itemize}
 
 			% TODO add diagram!
-			% \includegraphics[width=\textwidth]{expandable_models}
-			\includegraphics[height=\textheight]{expandable_models}
+			\includegraphics[height=\textheight]{expandable_models_simple}
 
 		\subsection{Generation models}
 			The system requires the generation of models. Generating all models based on one single model would decrease the complexity of the system, but it would not guarantee success.
@@ -143,6 +142,7 @@
 			Since the AutoML approach would be more computational intensive, it would be less desirable to run. Therefore, the approach would be for the database search to happen first, where known possibly good models would be first tested. If a good model is found, then the search stops and if no model is found, the system would resort to AutoML to find a suitable model.
 
 			% TODO add diagram
+			\includegraphics[height=\textheight]{expandable_models_generator}