Add categories to schema