Fix missing highlighting in some code examples