Create HW_1.py