The homework for python learning