1 files changed, 7 insertions, 1 deletions
diff --git a/ee1301/wk1/hw1_directory/strap012_1B.cpp b/ee1301/wk1/hw1_directory/strap012_1B.cpp
index 472f96d..02fd560 100644
--- a/ee1301/wk1/hw1_directory/strap012_1B.cpp
+++ b/ee1301/wk1/hw1_directory/strap012_1B.cpp
@@ -9,11 +9,17 @@ int main()
   cin >> option;
   
   if (option == "i") {
+    
     cout << "E";
+    
     }else{ if (option == "f") {
+      
       cout << "e";
+      
       }else{
-          cout << "nah.";
+        
+          cout << "Invalid option.";
       }
   }
+  cout << endl;
 }
 \ No newline at end of file