diff --git a/surveys/flame_graph/README.md b/surveys/flame_graph/README.md
index c3243b2..d0ac1fa 100644
--- a/surveys/flame_graph/README.md
+++ b/surveys/flame_graph/README.md
@@ -8,14 +8,16 @@ fn add(a, b) {
     + a b
 }
 
-let result = 0;
-let i = 0;
-loop {
-    if >= i 10 {
-        break;
+fn main() {
+    let result = 0;
+    let i = 0;
+    loop {
+        if >= i 10 {
+            break;
+        }
+        result = add(result, 5);
+        i = + i 1;
     }
-    result = add(result, 5);
-    i = + i 1;
 }
 ```