Updated and added category model