diff --git a/du6/program.c b/du6/program.c
index c8999e9..5197b9b 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -86,17 +86,20 @@ int main(void){
          }
 
     }
-    //if(!whitespace && tree != NULL){
-        //printf("No whitespace after database\n");
-       // return 0;
-    //}
     int numOfLeaves = count_leaves(tree);
     char answer[SIZE];
+    char existence_line[SIZE];
+    bool exs = false;
+    fgets(existence_line, SIZE, stdin);
+    if(existence_line[0]){
+        bool exs = true;
+    }
     printf("Expert z bufetu to vie.\n");
-    if(!tree || !whitespace){
+    if(!tree || !whitespace || !exs){
         printf("Chybna databaza\n");
         return 0;
     }
+
     printf("Pozna %d druhov ovocia a zeleniny.\n", numOfLeaves);
     printf("Odpovedajte 'a' pre prvu moznost alebo 'n' pre druhu moznost.");
     while(tree){