Add highlights for YARD