diff --git a/du5/program.c b/du5/program.c
index b556e90..b9eb1a6 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -54,7 +54,9 @@ void spustac(struct uzol* uzol) {
         spustac (uzol -> lavy);
     } else if (odpoved == 'n') {
         spustac (uzol -> pravy);
-    } else { // uvidim co test ocakava ked dam zle odpoved
+    } else if (odpoved == '\n'){ 
+        printf ("Koniec vstupu\n");
+    } else { 
         printf ("Nerozumiem\n");
     }