Added problem set 2 program bonsaikon.py