Create homework1.py