diff --git a/source/_static/code/fibonacci_iteration.py b/source/_static/code/fibonacci_iteration.py
index bb8ee8d878916c47c9500adb357f65ce67381b9c..673d5c381bb7b4f735732658af4addcf1cedf9bf 100644
--- a/source/_static/code/fibonacci_iteration.py
+++ b/source/_static/code/fibonacci_iteration.py
@@ -1,13 +1,18 @@
 
-fib_suite = []
-n = 0
-while n < 10:
-    if n == 0:
-        fib_suite.append(0)
-    elif n == 1:
-        fib_suite.append(1)
-    else: 
-        res = fib_suite[n-1] + fib_suite[n-2] 
-        fib_suite.append(res)
-    n += 1
-print ', '.join([str(i) for i in fib_suite])
\ No newline at end of file
+def fibonacci(n):
+    fib_suite = []
+    i = 0
+    while i <= n:
+        if i == 0:
+            fib_suite.append(0)
+        elif i == 1:
+            fib_suite.append(1)
+        else: 
+            res = fib_suite[i-1] + fib_suite[i-2] 
+            fib_suite.append(res)
+        i += 1
+    return fib_suite
+
+print ', '.join([str(i) for i in fibonacci(10)])
+
+