Square() creating function; Python Fund I, Functions Fund5

Screen Link: Learn About Functions — Parameters And Arguments | Dataquest

My Code:

def square(a_number):
    # Variable assignment step
    squared_number = a_number * a_number 
    return squared_number

So, I have what could be a pretty straightforward question. In the above example, the return statement could be whatever I want? In other words, I imagine that I get to decide in the body what I want this squaring action to equal to (squared_number), so long as the name I choose is the same in my return statement.
Is this logic correct? I have tried with different code using this logic and so far seems to work. Thanks for the help

That’s correct. You are defining/creating the function so you decide what it should do. You can have it return anything, but, of course, since you are creating a function to square a number it needs to do that.