Correct code highlight

Fixes code highlighting problems.

If there is a unwated change, just tell me, I'll remove it.