Enhance the "Edit a Category" logic