Add tests for iterators