Fix code highlighting issues