diff --git a/calculator.py b/calculator.py index c4cfd16..d18f00f 100644 --- a/calculator.py +++ b/calculator.py @@ -16,6 +16,8 @@ elif choice == 3: print("Output: ", num1*num2) elif choice == 4: + if num1 == 0 or num2 == 0: + exit() print("Output: ", num1/num2) elif choice == 5: print("Exiting...")