Adding multiplication

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()