diff --git a/du2/program.c b/du2/program.c
index f8df7bf..dd178a8 100644
--- a/du2/program.c
+++ b/du2/program.c
@@ -3,7 +3,7 @@
 #include <stdlib.h>
 
 #define SIZE 100
- 
+
 struct student { 
 
     char meno[SIZE];
@@ -14,6 +14,7 @@ struct student {
 int najdi_studenta(struct student* studenti, int pocet, const char* meno) {
 
     for (int i = 0; i < pocet; i++) {
+
         if (strcmp(studenti[i].meno, meno) == 0) {
             return i;
         }
@@ -49,8 +50,8 @@ int main() {
         char* koniec = NULL;
         int hodnota = (int)strtol(riadok, &koniec, 10);
 
-        if (koniec == riadok) {
-            continue;
+        if (koniec == riadok || hodnota == 0) {
+            break;
         }
 
         while (*koniec == ' ') {
@@ -70,6 +71,7 @@ int main() {
         if (zaciatok_mena[velkost_mena - 1] == '\n') {
             velkost_mena--;
         }
+
         if (velkost_mena <= 0) {
             continue;
         }