diff --git a/06/tasks/task_04_factorial.py b/06/tasks/task_04_factorial.py
index 63bd358..3de2cab 100644
--- a/06/tasks/task_04_factorial.py
+++ b/06/tasks/task_04_factorial.py
@@ -1,7 +1,13 @@
 # task_04_factorial.py
 
+
 def factorial(n: int) -> int:
     """
     Vrátí faktoriál čísla n.
     """
+    if n == 0 or n == 1:
+        return 1
+    else:
+        return n * factorial(n-1)
+