Update cs1504.py