Update hw0_tes.py