Changed the controller to add the link