Added post-category assignment functionality