Fix some code samples in the doc to be correctly highlighted