homework for Learning Python