Update ans1.py