diff --git a/hw01/main.py b/hw01/main.py index 020d1d7..64ace1a 100644 --- a/hw01/main.py +++ b/hw01/main.py @@ -4,10 +4,13 @@ def add(x,y): def sub(x,y): return x-y +def mult(x,y): + return x*y + def calc(): while(True): - operation = input("Enter operation from [+,-]: ") - if operation =="+" or operation == "-": + operation = input("Enter operation from [+,-,*]: ") + if operation =="+" or operation == "-" or operation == "*": break print("Enter valid operation") @@ -20,6 +23,9 @@ def calc(): if operation == "-": print(sub(x,y)) + if operation == "*": + print(mult(x,y)) + if __name__ == "__main__": calc() \ No newline at end of file