diff --git a/du2/program.c b/du2/program.c
index fb8c447..8c61570 100644
--- a/du2/program.c
+++ b/du2/program.c
@@ -4,13 +4,19 @@
 int main(){
     int i;
     int pole[VELKOST_POLA];
-for(i=0; i<VELKOST_POLA; i++){
-    int value = 0;
+    int max=0;
+    int value=0;
+while(i<VELKOST_POLA){
     printf('Zadaj hodnotu: ');
     int r = scanf("%d",&value);
     pole[i]=r;
+    if(r==1){
+        i++;
     }
-
+    else{
+        break;
+    }
+}
 
 return 0;
 }