updated control.py