python generator solution to problem 2
Merge branch 'python002'