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]);