python hw2. done