added in the categories functionality