Update bonus1.py