got category logic working