python hw4. first two