diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
index ee2e7c30..50c2c873 100644
--- a/vision_agent/tools/meta_tools.py
+++ b/vision_agent/tools/meta_tools.py
@@ -6,7 +6,6 @@
 import tempfile
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Union
-from uuid import UUID
 
 from IPython.display import display
 
@@ -105,13 +104,14 @@ def load(self, file_path: Union[str, Path]) -> None:
 
     def show(self) -> str:
         """Shows the artifacts that have been loaded and their remote save paths."""
-        out_str = "[Artifacts loaded]\n"
+        output_str = "[Artifacts loaded]\n"
         for k in self.artifacts.keys():
-            out_str += (
+            output_str += (
                 f"Artifact {k} loaded to {str(self.remote_save_path.parent / k)}\n"
             )
-        out_str += "[End of artifacts]\n"
-        return out_str
+        output_str += "[End of artifacts]\n"
+        print(output_str)
+        return output_str
 
     def save(self, local_path: Optional[Union[str, Path]] = None) -> None:
         save_path = (
@@ -237,7 +237,7 @@ def edit_code_artifact(
     new_content_lines = [
         line if line.endswith("\n") else line + "\n" for line in new_content_lines
     ]
-    lines = artifacts[name].splitlines()
+    lines = artifacts[name].splitlines(keepends=True)
     edited_lines = lines[:start] + new_content_lines + lines[end:]
 
     cur_line = start + len(content.split("\n")) // 2
@@ -274,6 +274,7 @@ def edit_code_artifact(
             )
 
             error_msg += f"\n[This is how your edit would have looked like if applied]\n{edited_view}\n\n[This is the original code before your edit]\n{original_view}"
+            print(error_msg)
             return error_msg
 
     artifacts[name] = "".join(edited_lines)
@@ -478,6 +479,16 @@ def use_florence2_fine_tuning(
     --------
         >>> diff = use_florence2_fine_tuning(artifacts, "code.py", "phrase_grounding", "23b3b022-5ebf-4798-9373-20ef36429abf")
     """
+
+    task_to_fn = {
+        "phrase_grounding": "florence2_phrase_grounding"
+    }
+
+    if name not in artifacts:
+        output_str = f"[Artifact {name} does not exist]"
+        print(output_str)
+        return output_str
+
     code = artifacts[name]
     if task.lower() == "phrase_grounding":
         pattern = r'florence2_phrase_grounding\((".*?", .*?)\)'
@@ -489,6 +500,12 @@ def replacer(match):
         raise ValueError(f"Task {task} is not supported.")
 
     new_code = re.sub(pattern, replacer, code)
+
+    if new_code == code:
+        output_str = f"[Fine tuning task {task} function {task_to_fn[task]} not found in code]"
+        print(output_str)
+        return output_str
+
     artifacts[name] = new_code
 
     diff = get_diff(code, new_code)