add logic to category