Add updating categories functionality