Allow using return in functions