Updated Edit method to use new util function.

I updated the edit method to take advantage of the new utility function.