diff --git a/du6/program.c b/du6/program.c
index 0bfe8d5..efd61cc 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -83,12 +83,16 @@ void freeTree(Node *x) {
 // hlavná funkcia
 int main() {
     Node *root = readTree();          // načítame strom zo vstupu
-    
+     printf("Expert z bufetu to vie.\n"); // vždy vypíšeme tento riadok
+
+    if (!root) {                    // ak strom je prázdny alebo chybne načítaný
+        printf("Chybna databaza\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)
-
-    printf("Expert z bufetu to vie.\n");       
+     
     printf("Pozna %d druhov ovocia a zeleniny.\n", k);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");