add the category and change the cms_model function