Update check_homework.py