diff --git a/src/app/main.c b/src/app/main.c
index 9b4f65b..8477854 100644
--- a/src/app/main.c
+++ b/src/app/main.c
@@ -4,8 +4,11 @@
 
 int main(void)
 {
+    //SET FLAGS
+    setFlags((inputFlags) {1, 1, 0});
+
     //RUN ENCRYPTION WITH TEXT_TO_ENCRYPT, GENERATE NEW KEY AND DO NOT CHECK FOR ACTIVE VERSION & PREVENT ANY OUTPUT
-    outputFlags encryptedText = encryptText(TEXT_TO_ENCRYPT, NULL, (inputFlags) {1, 1, 0});
+    outputFlags encryptedText = encryptText(TEXT_TO_ENCRYPT, NULL);
 
     //SIMPLE TEXT
     printf
diff --git a/src/lib/test/main.c b/src/lib/test/main.c
index c61aca4..c55ea80 100644
--- a/src/lib/test/main.c
+++ b/src/lib/test/main.c
@@ -20,6 +20,9 @@ int main(void)
         0 //UPDATE
     };
 
+    //SET FLAGS
+    setFlags(flags);
+
     //SET KEY_LENGTH TO 100
     setKeyLength(100);
     keyBuffer = malloc(getKeyLength() + 1);
@@ -28,7 +31,7 @@ int main(void)
     setEncryptionSeparator('|');
 
     //ENCRYPT
-    outputFlags encrypted = encryptText(TEST_TEXT, NULL, flags);
+    outputFlags encrypted = encryptText(TEST_TEXT, NULL);
 
     strcpy(textBuffer, encrypted.outputText); //GET ENCRYPTED TEXT
     strcpy(keyBuffer, encrypted.usedKey); //GET KEY
@@ -38,7 +41,7 @@ int main(void)
     deallocateOutput(encrypted);
 
     //DECRYPT
-    encrypted = decryptText(textBuffer, keyBuffer, flags);
+    encrypted = decryptText(textBuffer, keyBuffer);
 
     timeBuffer += encrypted.elapsedTime; //GET TIME 1