Add edit category logic