Add documentation highlight for code