Update CSC291_1.py