Add custom category comment support