Adding an import for categories