diff --git a/du5/program.c b/du5/program.c
index bb005b1..b42bb3f 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -130,21 +130,18 @@ int main()
     //zisk. korena
     int root = buildATree();
 
-    if (tree[root].yes == -1 && tree[root].no == -1)
-    {
-        printf("Koniec vstupu.\n");
-        return 0;
-    }
-
     printf("Expert z bufetu to vie.\n");
-    if (answerCount != 0) //vypise vysledky/statistiky, AK nebola chyba pri citani odpovedi
+    if (answerCount == 0)
     {
-        printf("Pozna %d druhov ovocia a zeleniny.\n", answerCount);
-        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+        printf("Chybna databaza\n");
     }
     else
     {
-        printf("Chybna␣databaza");
+        printf("Expert z bufetu to vie.\n");
+        printf("Pozna %d druhov ovocia a zeleniny.\n", answerCount);
+        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+
+        goThroughTheTree(root);
     }
     
     goThroughTheTree(root);