Fix code highlighting problems