feat(generators): Adding support for generator functions