Add category tests