Refine the run code example to support the controller.