Update treehouse.py