Fix review comments:

- put utility function creation at the beginning of "update step"
- explain where we add edit button

Author: ttypic

src/pages/docs/chat/getting-started/kotlin.mdx

@@ -418,15 +418,26 @@ You'll see the message in your app's chat box UI. If you have sent a message via
 
 If your client makes a typo, or needs to update their original message then they can edit it. To do this, you can extend the functionality of the `ChatBox` component to allow updating of messages.
 
-1. Add the edited state variable:
+1. We will begin by adding this utility function to facilitate message data updates in the UI:
+
+ <Code>
+ ```kotlin
+ inline fun <T> MutableList<T>.replaceFirstWith(replacement: T, predicate: (T) -> Boolean) {
+     val index = indexOfFirst(predicate)
+     if (index != -1) set(index, replacement)
+ }
+ ```
+ </Code>
+
+2. Add the edited state variable:
 
 <Code>
 ```kotlin
 var edited: Message? by remember { mutableStateOf(null) }
 ```
 </Code>
 
-2. Update the message subscription to handle edited messages:
+3. Update the message subscription to handle edited messages:
 
 <Code>
 ```kotlin
@@ -448,7 +459,10 @@ DisposableEffect(room) {
 ```
 </Code>
 
-3. Add an edit button to each message:
+4. Let's enhance the message display. To add an edit button to each message, we'll first need to locate
+the `Row` composable function within your layout. This is the component responsible for rendering
+the message content itself, along with the sender's username and the message timestamp.
+Once we've identified this `Row`, we will integrate the new edit button alongside these existing elements:
 
 <Code>
 ```kotlin
@@ -487,7 +501,7 @@ Row(
 ```
 </Code>
 
-4. Update the send button to handle both new messages and edits:
+5. Update the send button to handle both new messages and edits:
 
 <Code>
 ```kotlin
@@ -521,18 +535,6 @@ Button(
 </Code>
 
 
-
-Add this utility function to help with message updates:
-
-<Code>
-```kotlin
-inline fun <T> MutableList<T>.replaceFirstWith(replacement: T, predicate: (T) -> Boolean) {
-    val index = indexOfFirst(predicate)
-    if (index != -1) set(index, replacement)
-}
-```
-</Code>
-
 When you click on the edit button in the UI, you can modify the text and it will send the updated message contents to the room.
 
 ## Step 6: Message history and continuity <a id="step-6"/>