range in lua using coroutine

range.lua

function range(n)
    return coroutine.wrap(function() 
        for i = 1, n, 1 do
            coroutine.yield(i)
        end
    end)
end

function main() 
    for i in range(10) do
        print(i)
    end
end

main()