Homework for Python 3200K