diff --git a/du6/program.c b/du6/program.c
index 9ff4562..0bfe8d5 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -45,11 +45,7 @@ int count(Node *x) {
 
 // funkcia pre spustenie chatbota
 void run(Node *x) {
-    if (!x) {                     // ak je uzol prázdny
-    printf("Koniec vstupu\n"); // vypíšeme koniec vstupu
-    return;
-}
-
+    if (!x) return;                   // prázdny uzol - nič nerobíme
 
     if (x->leaf) {                    // ak sme na liste (odpoveď)
         printf("*%s\nKoniec\n", x->t);  // vypíšeme odpoveď a koniec
@@ -59,7 +55,7 @@ void run(Node *x) {
     printf("%s\n", x->t);             // vypíšeme otázku
 
     if (!fgets(s, MAX, stdin)) {      // načítame odpoveď od používateľa
-        printf("Nerozumiem\n");  // ak nič, upozorníme
+        printf("Koniec vstupu\n");  // ak nič, upozorníme
         return;
     }
 
@@ -87,12 +83,7 @@ void freeTree(Node *x) {
 // hlavná funkcia
 int main() {
     Node *root = readTree();          // načítame strom zo vstupu
-
-    if (!root) {                    // ak strom je prázdny
-        printf("Koniec vstupu\n");
-        return 0;
-    }
-
+    
     fgets(s, MAX, stdin);             // prečítame prázdny riadok medzi databázou a vstupom používateľa
 
     int k = count(root);              // spočítame počet listov (druhov tovaru)