Implement custom iterator
Update docs