Correct code highlighting