fix: indentation + add parameters

app.py

@@ -16,7 +16,7 @@ def my_fairytale_tool(arg:int)-> str: #it's import to specify the return type
         arg: random number
     """
 
-        fairytales = [
+    fairytales = [
         "Cinderella",
         "Snow White",
         "Little Red Riding Hood",
@@ -25,14 +25,11 @@ def my_fairytale_tool(arg:int)-> str: #it's import to specify the return type
     ]
     
     # Randomly select  fairytales from the list
-    selected_1 = fairytales[int(arg / 5)]
-    fairytales.pop(int(arg / 4))
-    selected_2 = fairytales[random.randint(0,4)]
+    selected_1 = fairytales[int(arg / len(fairytales))]
+    fairytales.pop(int(arg / len(fairytales)))
+    selected_2 = fairytales[random.randint(0,len(fairytales)-1)]
 
-
-    
     return f"What a crazy idea to see a story of {selected_1} vs {selected_2}"
-    #return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: