diff --git a/a4/program.c b/a4/program.c
index 347828b..4b04bd0 100644
--- a/a4/program.c
+++ b/a4/program.c
@@ -43,11 +43,7 @@ int main() {
     int n = 0;
 
     // Načítanie vstupu
-    // printf("Zadajte čísla oddelené medzerou: ");
-    // if (!fgets(input, sizeof(input), stdin)) {
-        // printf("Chyba pri načítaní vstupu.\n");
-        // return 1;
-    // }
+  
 
     // Parsovanie vstupu
     char *token = strtok(input, " ");
@@ -56,6 +52,13 @@ int main() {
         token = strtok(NULL, " ");
     }
 
+    // Ak je na vstupe len jedno číslo
+    if (n == 1) {
+        printf("Je to taka kopa:\n");
+        printf("%d\n", arr[0]);
+        return 0;
+    }
+
     // Overenie, či je to binárna kopa
     if (is_min_heap(arr, n)) {
         printf("Je to taka kopa:\n");