You now have a for loop so I think you can remove the TODO :)