Return variable from function on exercise 001