generators in python exercises