From 17fa99cf7f97d9c2b21c961fe6b25dbc5735a13b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Vl=C4=8Dek?= <tomas.vlcek@student.tuke.sk>
Date: Tue, 14 Apr 2026 23:24:05 +0000
Subject: [PATCH] Aktualizovat du5/program.c

---
 du5/program.c | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/du5/program.c b/du5/program.c
index 9e0a5ab..57e7d0d 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -40,8 +40,15 @@ int main()
         }
         else if (line[1] == '\n')
         {
-            answer = line[0];
-            break;
+            if (answer == '0')
+            { 
+                answer = line[0]; 
+            }
+            else
+            {
+                answer2 = line[0];
+            }
+            
         }
         else if (!isFirstQuestionRead)
         {