need to add check for each category