Add category edit function