Jconnor get units as generator

Leveraging the new get_units as_generator support