Added missing 'edit' method to fix update feature