Fix code highlights