Working on adding a controller.