Add controller.go