python HW3.py