Update controller.py