Add highlighting for code samples