From 437b0b7d1977192cfa702ce7f744dd734c4da27b Mon Sep 17 00:00:00 2001
From: Maryna Kravtsova <mk757ye@omega.tuke.sk>
Date: Sat, 21 Nov 2020 16:32:04 +0100
Subject: [PATCH] buffet

---
 cv8/program.c | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/cv8/program.c b/cv8/program.c
index 8a25fe7..58bbf43 100644
--- a/cv8/program.c
+++ b/cv8/program.c
@@ -101,12 +101,14 @@ int count_l(struct tree* node){
 
 
 int main(){
-    printf("Expert z bufetu to vie\n");
-    printf("Pozna 2 druhov ovocia a zeleniny\n");
-    printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
     struct tree* tree = NULL;
-              
     tree = read_tree();
+    int count = count_l(tree);
+
+    printf("Expert z bufetu to vie\n");
+    printf("Pozna %d druhov ovocia a zeleniny\n", count);
+    printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.\n");
+      
     //print_tree(tree, 3);
     //display(tree);
     destroy_tree(tree);