Move a loop into a function