Adding correct categories