Adding code for three categories