Add censor function