diff --git a/lollms/databases/discussions_database.py b/lollms/databases/discussions_database.py
index 814421f..26920d1 100644
--- a/lollms/databases/discussions_database.py
+++ b/lollms/databases/discussions_database.py
@@ -911,7 +911,7 @@ class Discussion:
                     pth = str(view_file).replace("\\","/").split('/')
                     if "discussion_databases" in pth:
                         pth = discussion_path_to_url(view_file)
-                        self.lollms.personality.new_message()
+                        self.lollms.personality.new_message("")
                         output = f'<img src="{pth}" width="800">\n\n'
                         self.lollms.personality.set_message_html(output)
                         self.lollms.close_message(client.client_id if client is not None else 0)
diff --git a/lollms/function_call.py b/lollms/function_call.py
index fa87105..0e48fc8 100644
--- a/lollms/function_call.py
+++ b/lollms/function_call.py
@@ -51,5 +51,4 @@ class FunctionCall:
         return constructed_context
         
     def process_output(self, context, llm_output:str):
-        if self.function_type == FunctionType.CONTEXT_UPDATE:
-            raise NotImplementedError("Subclasses must implement the process_output for CONTEXT_UPDATE functions.")
\ No newline at end of file
+        return llm_output
\ No newline at end of file
diff --git a/lollms/prompting.py b/lollms/prompting.py
index 444609a..5940804 100644
--- a/lollms/prompting.py
+++ b/lollms/prompting.py
@@ -156,6 +156,8 @@ class LollmsContextDetails:
         append_context("think_first_mode")
         
         append_context("extra")
+        append_context("discussion_messages", template.system_custom_header("Discussion")+"\n")
+        
         found_classic_function = False
         if not ignore_function_calls:
             for function_call in self.function_calls:
@@ -191,7 +193,6 @@ class LollmsContextDetails:
                     ])
                 )              
 
-        append_context("discussion_messages", template.system_custom_header("Discussion")+"\n")
         # Add custom entries if provided
         if custom_entries:
             full_context+=custom_entries