Updated Controller.cpp