Added update in controller