diff --git a/a2/program.c b/a2/program.c
index 88605a6..9c839a3 100644
--- a/a2/program.c
+++ b/a2/program.c
@@ -57,11 +57,11 @@ void print_heap_with_levels(struct osoba** heap, int size, int index, int depth)
     
 }
 int main(){
-    struct osoba* pole[20];
+    struct osoba* pole[30];
     memset(pole,0, sizeof(struct osoba*));
     int pocet = 0;
-      char buff[20];
-        if(!fgets(buff, 20, stdin)){
+      char buff[100];
+        if(!fgets(buff, 100, stdin)){
             printf("Chyba\n");
             return 0;
         }
@@ -83,22 +83,19 @@ int main(){
 
          
          
-       
-         
+    for(int i =0; i<pocet; i++)   {
+        printf("%d\n", pole[i]->vek);
+    }
       int i = 0;       
-int l = i*2 +1;
-    int r = i*2 +2;
-    while (l< pocet && r<pocet && i<pocet)
+    while (2*i+1< pocet && 2*i+2<pocet)
 
     { 
       
-        if(pole[l]->vek <pole[i]->vek || pole[r]->vek <pole[i]->vek){
+        if(pole[2*i+1]->vek <pole[i]->vek || pole[2*i+2]->vek <pole[i]->vek){
             printf("Nie je kopa.\n");
             return 0;
         }
         i++;
-         l = i*2 +1;
-    r = i*2 +2;
 
     }
     printf("Je to taka kopa:\n");