Add Product and Category class to models