diff --git a/03_recursion/python/06_find_max.py b/03_recursion/python/06_find_max.py
index 9080084..bb5afc5 100644
--- a/03_recursion/python/06_find_max.py
+++ b/03_recursion/python/06_find_max.py
@@ -1,5 +1,9 @@
 def find_max(arr):
-    if len(arr) == 2:
+    if len(arr) == 0:
+        return 0
+    elif len(arr) == 1:
+        return arr[0]
+    elif len(arr) == 2:
         return arr[0] if arr[0] > arr[1] else arr[1]
     sub_max = find_max(arr[1:])
-    return arr[0] if arr[0] > sub_max else sub_max
\ No newline at end of file
+    return arr[0] if arr[0] > sub_max else sub_max