Fix code block highlighting