diff --git a/cv7/program.c b/cv7/program.c
index 861bb24..e56d596 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -15,6 +15,7 @@ struct tree* read_tree(int* counter) {
     memset(buffer, 0, SIZE);
     char* r = fgets(buffer, SIZE, stdin);
     if(r == NULL){
+        printf("Expert z bufetu to vie.\n");
         printf("Chybna databaza\n");
         return 0;
     }
@@ -65,11 +66,12 @@ void print_tree(struct tree* tree) {
 int main() {
     int counter = 0;
     struct tree* root = read_tree(&counter);
-    printf("Expert z bufetu to vie.\n");
-
-    printf("Pozna %d druhov ovocia a zeleniny.\n", counter);
-    printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
-    print_tree(root);
+    if(root != 0){
+        printf("Expert z bufetu to vie.\n");
+        printf("Pozna %d druhov ovocia a zeleniny.\n", counter);
+        printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+        print_tree(root);
+    }
     destroy_tree(root);
     return 0;
 }
\ No newline at end of file