diff --git a/ra_aid/__main__.py b/ra_aid/__main__.py
index e01eb6c..bd0b307 100644
--- a/ra_aid/__main__.py
+++ b/ra_aid/__main__.py
@@ -481,6 +481,11 @@ Be very thorough in your research and emit lots of snippets, key facts. If you t
                     expert_enabled=expert_enabled
                 )
         except TaskCompletedException:
+            console.print(Panel(
+                "[green]Task was completed successfully as a one-shot operation.[/green]",
+                title="Task Completed",
+                style="green"
+            ))
             sys.exit(0)
     finally:
         pass
diff --git a/ra_aid/env.py b/ra_aid/env.py
new file mode 100644
index 0000000..e69de29
diff --git a/ra_aid/prompts.py b/ra_aid/prompts.py
index 274e590..8970619 100644
--- a/ra_aid/prompts.py
+++ b/ra_aid/prompts.py
@@ -65,22 +65,7 @@ No Planning or Problem-Solving
 
 You must remain strictly within the bounds of describing what currently exists.
 
-Single-Shot Task Detection
-
-    Autonomously determine if a task can be completed immediately without further planning:
-        - Simple informational queries that can be answered directly from research
-        - Basic implementation tasks that don't require complex changes
-        - Cases where further planning would not add value
-        - Situations where immediate response meets all user requirements
-
-    One-shot completion will be blocked if:
-        - Research subtasks have been requested
-        - Complex implementation has been explicitly requested
-
-    If you determine a task can be completed in a single shot:
-        1. Complete the necessary research or basic implementation work
-        2. Document your findings using emit_research_notes
-        3. Call one_shot_completed() to immediately conclude the task
+If this is a simple task that can be completed in one shot, do the change using tools available and call one_shot_completed.
     
     Only use single-shot completion for truly straightforward tasks that don't require additional planning or extensive changes.
     If the change is estimated to be less than 100 lines and less than 5 files, it is definitely a single-shot task.
diff --git a/tests/ra_aid/test_env.py b/tests/ra_aid/test_env.py
new file mode 100644
index 0000000..e69de29