Create homework.py