Update whole.py