Enhanced the edit button to work.