Always pass the context through
Rename to_return variable to make more sense