diff --git a/du1/program.c b/du1/program.c
index 7cb165e..60a051f 100644
--- a/du1/program.c
+++ b/du1/program.c
@@ -3,23 +3,31 @@
 int main(){
 	int c = 0;
 	int a = 0;
+	int counter = 0;
 	char small[] = "abcdefghijklmnopqrstuvwxyz";
     	char big[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
 	char d;
-	while( c != EOF){
+	while(c = 1){
 		c = getchar();
 		d = c;
 		int a = 0;
+		if(d == '\n'){
+			counter++;
+			break;
+		}
 		while(a > 26){
 			if(d == small[a]){
-				c = big[a];
+				d = big[a];
 				break;
 			}else if(d == big[a]){
-				c = small[a];
+				d = small[a];
 				break;
-			}		
+			}
+			a++;		
 		}
+		c = d;
 		putchar (c);
 	}
+	printf("\nPočet riadkov: %d\n",counter);
 	return 0;
 }