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