diff --git a/ex01/main.cpp b/ex01/main.cpp
index 2c7281d..d800766 100644
--- a/ex01/main.cpp
+++ b/ex01/main.cpp
@@ -40,7 +40,10 @@ int	main()
 			if (new_contact.isValid())
 				phone_book.add_contact(new_contact);
 			else
+			{
 				std::cerr << "Contact Error !" << std::endl;
+				return (1);
+			}
 		}
 		else if (command == "SEARCH")
 		{
@@ -50,6 +53,11 @@ int	main()
 		{
 			return (0);
 		}
+		else
+		{
+			std::cerr << "Command not found" << std::endl;
+			return (2);
+		}
 	}
 	return (0);
 }