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
print(square(6))
```

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