Add support for controller in teleopMain