Homework1-2.py