diff --git a/du5/program.c b/du5/program.c
index 944e940..94318d5 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -32,12 +32,7 @@ int main()
                 sscanf(line, "%s [^\n]", &option2);
             }
         }
-        else if (line[0] == 'n' && line[1] == '\n')
-        {
-            answer = line[0];
-            break;
-        }
-        else if (line[0] == 'a' && line[1] == '\n')
+        else if (line[1] == '\n')
         {
             answer = line[0];
             break;
@@ -58,10 +53,14 @@ int main()
     {
         printf("%s\n", option2);
     }
-    else
+    else if (answer == '0')
     {
         printf("Nerozumiem\n");
     }
+    else
+    {
+        printf("Koniec␣vstupu");
+    }
     if (!err) { printf("Koniec\n"); }
 
     return 0;