Better handling of generators/iterators (supports arguments)