Update controller.lua