python hw4. done