fix(complexity): Adding support for generator functions