add homework.py