Finished add prime category function