add return in classify_comment