diff --git a/documentation/figures/class_diagram.png b/documentation/figures/class_diagram.png
index bfa40e2..076631a 100644
Binary files a/documentation/figures/class_diagram.png and b/documentation/figures/class_diagram.png differ
diff --git a/documentation/figures/class_diagram_new.png b/documentation/figures/class_diagram_new.png
new file mode 100644
index 0000000..369ab41
Binary files /dev/null and b/documentation/figures/class_diagram_new.png differ
diff --git a/documentation/figures/system_architecture_simplified.png b/documentation/figures/system_architecture_simplified.png
new file mode 100644
index 0000000..2b49f0d
Binary files /dev/null and b/documentation/figures/system_architecture_simplified.png differ
diff --git a/documentation/figures/uml_diagrams.drawio b/documentation/figures/uml_diagrams.drawio
index aa615ee..8618cfc 100644
--- a/documentation/figures/uml_diagrams.drawio
+++ b/documentation/figures/uml_diagrams.drawio
@@ -1,6 +1,6 @@
 <mxfile host="65bd71144e" scale="10" border="0">
     <diagram id="2JWEvkMf0evVfgLf1cgb" name="Class">
-        <mxGraphModel dx="1282" dy="1002" grid="1" gridSize="10" guides="1" tooltips="1" connect="1" arrows="1" fold="1" page="1" pageScale="1" pageWidth="827" pageHeight="1169" math="0" shadow="0">
+        <mxGraphModel dx="1285" dy="1002" grid="1" gridSize="10" guides="1" tooltips="1" connect="1" arrows="1" fold="1" page="1" pageScale="1" pageWidth="827" pageHeight="1169" math="0" shadow="0">
             <root>
                 <mxCell id="0"/>
                 <mxCell id="1" parent="0"/>
@@ -10,7 +10,7 @@
                 <mxCell id="McSh0blU3mZ8xUULmvc6-2" value="&lt;p style=&quot;margin:0px;margin-top:4px;text-align:center;&quot;&gt;&lt;i&gt;&amp;lt;&amp;lt;Interface&amp;gt;&amp;gt;&lt;/i&gt;&lt;br&gt;&lt;b&gt;IPromptManager&lt;/b&gt;&lt;/p&gt;&lt;hr size=&quot;1&quot;&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;+ generate_conversation_context(&lt;/p&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;&lt;span style=&quot;white-space: pre;&quot;&gt;&#9;&lt;/span&gt;conversation,&lt;br&gt;&lt;/p&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;&lt;span style=&quot;white-space: pre;&quot;&gt;&#9;&lt;/span&gt;agent_type,&lt;br&gt;&lt;/p&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;&lt;span style=&quot;white-space: pre;&quot;&gt;&#9;&lt;/span&gt;llm_type&lt;/p&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;): dict[]&lt;/p&gt;" style="verticalAlign=top;align=left;overflow=fill;fontSize=12;fontFamily=Helvetica;html=1;flipH=1;" parent="1" vertex="1">
                     <mxGeometry x="590" y="450" width="190" height="122" as="geometry"/>
                 </mxCell>
-                <mxCell id="McSh0blU3mZ8xUULmvc6-3" value="&lt;p style=&quot;margin:0px;margin-top:4px;text-align:center;&quot;&gt;&lt;i&gt;&amp;lt;&amp;lt;Interface&amp;gt;&amp;gt;&lt;/i&gt;&lt;br&gt;&lt;b&gt;IAssistant&lt;/b&gt;&lt;/p&gt;&lt;hr size=&quot;1&quot;&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;+ generate_response(&lt;span style=&quot;background-color: initial;&quot;&gt;conversation): str&lt;/span&gt;&lt;/p&gt;" style="verticalAlign=top;align=left;overflow=fill;fontSize=12;fontFamily=Helvetica;html=1;" parent="1" vertex="1">
+                <mxCell id="McSh0blU3mZ8xUULmvc6-3" value="&lt;p style=&quot;margin:0px;margin-top:4px;text-align:center;&quot;&gt;&lt;i&gt;&amp;lt;&amp;lt;Interface&amp;gt;&amp;gt;&lt;/i&gt;&lt;br&gt;&lt;b&gt;IAgent&lt;/b&gt;&lt;/p&gt;&lt;hr size=&quot;1&quot;&gt;&lt;p style=&quot;margin:0px;margin-left:4px;&quot;&gt;+ generate_response(&lt;span style=&quot;background-color: initial;&quot;&gt;conversation): str&lt;/span&gt;&lt;/p&gt;" style="verticalAlign=top;align=left;overflow=fill;fontSize=12;fontFamily=Helvetica;html=1;" parent="1" vertex="1">
                     <mxGeometry x="303" y="351" width="221" height="70" as="geometry"/>
                 </mxCell>
                 <mxCell id="McSh0blU3mZ8xUULmvc6-6" value="&lt;b&gt;SSHPythonRuntime&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
@@ -67,13 +67,13 @@
                         <mxPoint x="410" y="770" as="targetPoint"/>
                     </mxGeometry>
                 </mxCell>
-                <mxCell id="McSh0blU3mZ8xUULmvc6-20" value="&lt;b&gt;GPTAssistant&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
+                <mxCell id="McSh0blU3mZ8xUULmvc6-20" value="&lt;b&gt;GPTAgent&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
                     <mxGeometry x="270" y="270" width="120" height="50" as="geometry"/>
                 </mxCell>
-                <mxCell id="McSh0blU3mZ8xUULmvc6-21" value="&lt;b&gt;LLaMAAssistant&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
+                <mxCell id="McSh0blU3mZ8xUULmvc6-21" value="&lt;b&gt;LLaMAAgent&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
                     <mxGeometry x="360.5" y="210" width="120" height="50" as="geometry"/>
                 </mxCell>
-                <mxCell id="McSh0blU3mZ8xUULmvc6-22" value="&lt;b&gt;CodeLLaMAAssistant&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
+                <mxCell id="McSh0blU3mZ8xUULmvc6-22" value="&lt;b&gt;CodeLLaMAAgent&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
                     <mxGeometry x="450" y="270" width="120" height="50" as="geometry"/>
                 </mxCell>
                 <mxCell id="McSh0blU3mZ8xUULmvc6-23" value="" style="endArrow=block;dashed=1;endFill=0;endSize=12;html=1;exitX=0.454;exitY=1.02;exitDx=0;exitDy=0;exitPerimeter=0;entryX=0.25;entryY=0;entryDx=0;entryDy=0;" parent="1" source="McSh0blU3mZ8xUULmvc6-20" target="McSh0blU3mZ8xUULmvc6-3" edge="1">
@@ -97,7 +97,7 @@
                 <mxCell id="McSh0blU3mZ8xUULmvc6-26" value="&lt;b&gt;ZeroShotPromptManager&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
                     <mxGeometry x="607.5" y="610" width="155" height="50" as="geometry"/>
                 </mxCell>
-                <mxCell id="McSh0blU3mZ8xUULmvc6-28" value="&lt;b&gt;CoTPromptManager&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
+                <mxCell id="McSh0blU3mZ8xUULmvc6-28" value="&lt;b&gt;FewShotPromptManager&lt;/b&gt;" style="html=1;" parent="1" vertex="1">
                     <mxGeometry x="607.5" y="361" width="155" height="50" as="geometry"/>
                 </mxCell>
                 <mxCell id="McSh0blU3mZ8xUULmvc6-29" value="" style="endArrow=block;dashed=1;endFill=0;endSize=12;html=1;exitX=0.5;exitY=0;exitDx=0;exitDy=0;entryX=0.5;entryY=1;entryDx=0;entryDy=0;" parent="1" source="McSh0blU3mZ8xUULmvc6-26" target="McSh0blU3mZ8xUULmvc6-2" edge="1">