diff --git a/estimate.py b/estimate.py
index f9af731..11c4ea0 100644
--- a/estimate.py
+++ b/estimate.py
@@ -7,13 +7,20 @@ class Estimate:
         self.theta1 = 0
         self.get_thetas()
     def get_thetas(self):
-        with open(self.path, 'r') as file:
-            self.data = pd.read_csv(file)
-            self.theta0 = self.data["theta0"].iloc[0]
-            self.theta1 = self.data["theta1"].iloc[0]
+        try:
+            with open(self.path, 'r') as file:
+                self.data = pd.read_csv(file)
+                self.theta0 = self.data["theta0"].iloc[0]
+                self.theta1 = self.data["theta1"].iloc[0]
+        except:
+            print("! Warning, no data has been found")
     def estimate_price(self, mileage):
         return self.theta0 + (self.theta1 * mileage)
 
 estimate = Estimate()
-input =  int(input("Please enter the mileage of your car: "))
-print("The estimated price for a", input, "kilometers car is :", estimate.estimate_price(input))
+try:
+    input =  int(input("Please enter the mileage of your car: "))
+except:
+    print("! Wrongs args format")
+else:
+    print("The estimated price for a", input, "kilometers car is :", estimate.estimate_price(input))