iterator and generator in python