Add method for when a category flag passed