updated the controller file