Refactor controller - update operation