Update controller2.py