Create hw3_1.py