Update controller.go