Use correct Python2/3 compatibility iterator.
Merge pull request #18 from ShaheedHaque/fix_iterator3

Use correct Python2/3 compatibility iterator.