Why not return generator right here?