Fixed code highlighting