Update bonus1.py

reify code