diff --git a/backend/agent/gemini_prompt.py b/backend/agent/gemini_prompt.py
index f38d71e2..e20ebd83 100644
--- a/backend/agent/gemini_prompt.py
+++ b/backend/agent/gemini_prompt.py
@@ -768,20 +768,19 @@ Now let's look at transportation options within Japan:
 Based on the searches, I'm going to compile information to create your itinerary. First, let me mark some research tasks as complete and update our todo list:
 
 <function_calls>
-<invoke name="str_replace">
-<parameter name="file_path">todo.md</parameter>
-<parameter name="old_str">## Research Phase
-- [ ] Research flight options from Seattle to Japan
-- [ ] Research best cities/regions to visit based on interests (historical sites, hidden gems, cultural experiences)
-- [ ] Research transportation options within Japan
-- [ ] Research accommodation options within budget
-- [ ] Research proposal location options</parameter>
-<parameter name="new_str">## Research Phase
+<invoke name="edit_file">
+<parameter name="target_file">todo.md</parameter>
+<parameter name="instructions">I am marking the research tasks as complete in my todo list.</parameter>
+<parameter name="code_edit">
+// ... existing code ...
+## Research Phase
 - [x] Research flight options from Seattle to Japan
 - [x] Research best cities/regions to visit based on interests (historical sites, hidden gems, cultural experiences)
 - [x] Research transportation options within Japan
 - [ ] Research accommodation options within budget
-- [x] Research proposal location options</parameter>
+- [x] Research proposal location options
+// ... existing code ...
+</parameter>
 </invoke>
 </function_calls>
 
diff --git a/backend/agent/sample_responses/2.txt b/backend/agent/sample_responses/2.txt
index a84510cd..169e0b7f 100644
--- a/backend/agent/sample_responses/2.txt
+++ b/backend/agent/sample_responses/2.txt
@@ -7,8 +7,7 @@ First, I'll create a todo list to organize our work and make sure we cover all a
 <function_calls>
 <invoke name="create_file">
 <parameter name="file_path">todo.md</parameter>
-<parameter name="file_contents">
-# Japan Trip Planning Todo
+<parameter name="file_contents"># Japan Trip Planning Todo
 
 ## Research Phase
 - [ ] Research flight options from Seattle to Japan
@@ -41,8 +40,7 @@ First, I'll create a todo list to organize our work and make sure we cover all a
 - [ ] Ensure all interests are addressed
 - [ ] Verify budget alignment
 - [ ] Finalize proposal location recommendation
-- [ ] Proofread and finalize all documents
-</parameter>
+- [ ] Proofread and finalize all documents</parameter>
 </invoke>
 </function_calls>
 
@@ -138,6 +136,28 @@ Based on the searches, I'm going to compile information to create your itinerary
 </invoke>
 </function_calls>
 
+Now I'll also add some specific recommendations based on my research using AI-powered file editing:
+
+<function_calls>
+<invoke name="edit_file">
+<parameter name="target_file">todo.md</parameter>
+<parameter name="instructions">I am adding a new "Flight & Transportation Recommendations" section after the Research Phase with specific findings from the web searches</parameter>
+<parameter name="code_edit">
+// ... existing code ...
+- [x] Research proposal location options
+
+## Flight & Transportation Recommendations
+- **Flights**: Seattle to Tokyo flights range from $800-1,200 roundtrip
+- **JR Pass**: 14-day pass recommended for $435 (covers most transportation)
+- **Recommended Route**: Tokyo → Kyoto → Nara → Osaka for optimal cultural experience
+- **Local Transport**: IC cards (Suica/Pasmo) for city transportation
+
+## Itinerary Creation
+// ... existing code ...
+</parameter>
+</invoke>
+</function_calls>
+
 ## Creating Your Itinerary
 
 Now I'll start crafting your detailed 7-day itinerary based on the research. First, let me research accommodation options within your budget:
@@ -310,8 +330,9 @@ This itinerary is designed for a couple traveling from Seattle to Japan from Apr
 
 4. **Hidden Gems Focus**: Throughout the itinerary, we've included lesser-known spots alongside must-see attractions to provide a balanced experience.
 
-5. **Flexibility**: This itinerary includes built-in flexibility. If you find a place you love, you can easily spend more time there by adjusting the schedule.
-</create-file>
+5. **Flexibility**: This itinerary includes built-in flexibility. If you find a place you love, you can easily spend more time there by adjusting the schedule.</parameter>
+</invoke>
+</function_calls>
 
 Now I'll start creating the HTML travel handbook for your trip. First, let's set up the CSS for the handbook to ensure it has a clean, attractive design:
 
@@ -989,6 +1010,12 @@ Now let's check that our files are in place and update our todo list:
 - [x] Research accommodation options within budget
 - [x] Research proposal location options
 
+## Flight & Transportation Recommendations
+- **Flights**: Seattle to Tokyo flights range from $800-1,200 roundtrip
+- **JR Pass**: 14-day pass recommended for $435 (covers most transportation)
+- **Recommended Route**: Tokyo → Kyoto → Nara → Osaka for optimal cultural experience
+- **Local Transport**: IC cards (Suica/Pasmo) for city transportation
+
 ## Itinerary Creation
 - [x] Create detailed day-by-day itinerary
 - [x] Balance activities between major attractions and hidden gems
@@ -1009,11 +1036,11 @@ Now let's check that our files are in place and update our todo list:
 - [x] Test handbook functionality
 
 ## Final Review
-- [ ] Review itinerary for balance and pacing
-- [ ] Ensure all interests are addressed
-- [ ] Verify budget alignment
-- [ ] Finalize proposal location recommendation
-- [ ] Proofread and finalize all documents
+- [x] Review itinerary for balance and pacing
+- [x] Ensure all interests are addressed
+- [x] Verify budget alignment
+- [x] Finalize proposal location recommendation
+- [x] Proofread and finalize all documents
 </parameter>
 </invoke>
 </function_calls>
@@ -1070,4 +1097,4 @@ You can preview the HTML travel handbook through the link provided by the expose
 
 Is there anything specific you'd like me to adjust or explain in more detail about the itinerary or travel handbook?</parameter>
 </invoke>
-</function_calls>
+</function_calls>
\ No newline at end of file