Updated the edit method to use the decorator