python hw 3