Implement correct Iterator type