Update control.py