Refactored generators in TraveralVisitor.
Iterator is working. Got caught up in PEP 380.