Update Controller.py

Better alignment of controller