Calculator math.sqrt

[jefsummers]

print("Select operation.")
print("1.Add")
print("2.Subtract")
print("3.Multiply")
print("4.Divide")
print("5.Square root")
# Take input from the user
choice = input("Enter choice(1/2/3/4/5):")
num1 = int(input("Enter first number: "))
if choice != "5" :
    num2 = int(input("Enter second number: "))

Then add the code to handle square roots after the code for Divide.