Update squirrel.py