Integrated the categories functionality