Fixed code block highlighting