Update TocHW4.py