From 2424a630a8d85df0ed34954179680e69275cd352 Mon Sep 17 00:00:00 2001
From: Illia Korpan <illia.korpan@student.tuke.sk>
Date: Wed, 15 Apr 2026 09:33:44 +0200
Subject: [PATCH] 123

---
 du5/program.c | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/du5/program.c b/du5/program.c
index 7e217bb..4f4559c 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -121,27 +121,30 @@ int main(void) {
     char buf[128];
     while (cur) {
         if (cur->is_answer) {
-            printf("*%s\n", cur->text);
+            printf("* %s\n", cur->text);
             break;
         }
         printf("%s\n", cur->text);
         if (!fgets(buf, sizeof(buf), stdin)) {
-            printf("Nerozumiem\n");
-            return 0;
+            printf("Koniec vstupu\n");
+            break;
         }
         char *p = buf;
         while (*p && isspace((unsigned char)*p)) p++;
         char c = *p;
+        if (c == '\0' || c == '\n') {
+            printf("Koniec vstupu\n");
+            break;
+        }
         if (c == 'a' || c == 'A') {
             cur = cur->yes;
         } else if (c == 'n' || c == 'N') {
             cur = cur->no;
         } else {
             printf("Nerozumiem\n");
-            return 0;
+            break;
         }
     }
-    printf("Koniec\n");
 
     free_tree(root);
     for (int i = 0; i < lines_count; ++i) free(lines[i]);