Fixed code highlights