diff --git a/cv7/program.c b/cv7/program.c
index 5376a04..d6e2aaf 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -76,6 +76,7 @@ int main() {
     
     if (root == NULL) {
         printf("Chyba pri nacitani bazy pravidiel.\n");
+        printf("Koniec vstupu\n");  // Выводим сообщение о завершении ввода при ошибке
         return 1;
     }
 
@@ -85,10 +86,11 @@ int main() {
 
     if (!ask_question(root)) {
         free_tree(root);
+        printf("Koniec vstupu\n");  // Завершаем с этим сообщением при ошибке ввода
         return 0;  // Завершаем программу при неверном вводе.
     }
 
-    printf("Koniec\n");
+    printf("Koniec vstupu\n");  // Завершаем с этим сообщением при успешном завершении
     free_tree(root);
     return 0;
 }