Update controller.py

updating the name of import