Help me in solving PYCALC04 problem

My issue

it is returning that you forgot to update the values to be returned by user_choice
_

My code

``````def calculatorDisplay():

display = """Welcome to Calculator

Choose one operation:
2. Subtract
3. Exit
"""
return(display)

def calculatorFunction(user_choice):
if user_choice == 1:
a, b = user_input()
return f"The sum is: {output}"
elif user_choice == 2:
print("Let's perform subtraction")
a, b = user_input()
output = subtraction(a, b)
return f"The difference is: {output}"
else:
return("Exit the program")

# Update the user_input(), addition() and subtraction() functions
def user_input():
print("Give two numbers on two lines")
a = int(input("Number 1 is: "))
b = int(input("Number 2 is: "))
return a,b

return (a+b)

def subtraction(a, b):
return (a-b)

if __name__ == '__main__':
print(calculatorDisplay())
user_choice = int(input("Select the operation: "))
value = calculatorFunction(user_choice)
print(value)
``````

Learning course: Python Projects
Problem Link: Perform calculations Practice Problem in Python Projects - CodeChef