diff --git a/a3/program.c b/a3/program.c
index 9d2c101..8fb970e 100644
--- a/a3/program.c
+++ b/a3/program.c
@@ -4,15 +4,15 @@
 int main() {
 //int l = i * 2 + 1;
 //int r = i * 2 + 2;
-    int i,j,n,count;
-
+    int i;
     int *a = calloc(MAX, sizeof(int));
-        for(int i;i<10;i++){
-            scanf("%d",a[i]);
-        }
+    if(a[i]!=0)
+    for (i = 0; i < MAX; i++) {
+        scanf("%d", &a[i]);
+    }
     //m = buildmaxheap(a, n);
     printf("Je to taka kopa:\n");
-    for (i = 0; i < 10; i++) {
+    for (i = 0; i < MAX; i++) {
         printf("%d\n", a[i]);
     }