Please help me with the task: Calculator.

Hello, I’m working on this task:

Here is my code:


a = int(input("Enter your first number."))
b = int(input("Enter your second number"))

operation = input("add or multiply")
if operation == "add":
   print(a + b)
elif operation == "multiply":
   print(a * b)

Can anyone please help me to solve it?